package com.lollipop.programming.sort;

import lombok.extern.slf4j.Slf4j;

/**
 * @Author Lollipop
 * @Date 2025/4/7 10:12
 * @Description 快速排序V1
 */
@Slf4j
public class QuickSortV1 {
    // 测试快速排序
    public static void main(String[] args) {
        int[] array = {10, 7, 8, 9, 1, 5, 16, 3};
        int n = array.length;

        log.info("未排序数组:");
        printArray(array);

        quickSort(array, 0, n - 1);

        log.info("排序后数组:");
        printArray(array);
    }

    // 快速排序主方法
    public static void quickSort(int[] array, int low, int high) {
        if (low < high) {
            // pi 是分区索引，array[pi] 已经排好序
            int pi = partition(array, low, high);

            // 递归地对 pi 之前的和之后的部分进行排序
            quickSort(array, low, pi - 1);
            quickSort(array, pi + 1, high);
        }
    }

    // 分区方法
    private static int partition(int[] array, int low, int high) {
        // 选择最右边的元素作为枢轴
        int pivot = array[high];
        // 较小元素的索引
        int i = (low - 1);

        for (int j = low; j < high; j++) {
            // 如果当前元素小于或等于枢轴
            if (array[j] <= pivot) {
                i++;

                // 交换 array[i] 和 array[j]
                int temp = array[i];
                array[i] = array[j];
                array[j] = temp;
            }
        }

        // 交换 array[i+1] 和 array[high] (或枢轴)
        int temp = array[i + 1];
        array[i + 1] = array[high];
        array[high] = temp;

        return i + 1;
    }

    // 打印数组方法
    public static void printArray(int[] array) {
        int n = array.length;
        for (int j : array) {
            log.info("{} ", j);
        }
        log.info("\n");
    }
}
